import os.path

from ai import AiConstant
from ai.utils import utils_file
import sys

OUTPUT_DIR = AiConstant.OUTPUT_PATH + 'masr/'
MODEL_DIR = OUTPUT_DIR + 'model/'
VOCAB_DIR = OUTPUT_DIR + 'vocab/'
LOG_PATH = AiConstant.LOG_PATH + 'log/masr.log'
RECORD_SAVE_DIR = OUTPUT_DIR + 'record/'
PREHAND_SAVE_DIR = OUTPUT_DIR + 'prehand/'

DATA_DIR = AiConstant.DATA_PATH + 'asr_pytorch/data_thchs30/data/'
for file in [OUTPUT_DIR, MODEL_DIR, VOCAB_DIR, LOG_PATH, RECORD_SAVE_DIR, PREHAND_SAVE_DIR]:
    utils_file.makedir_for_file_or_dir(file)

logger = AiConstant.AI_LOGGER(LOG_PATH)


def get_logger():
    return logger


def get_now():
    return utils_file.get_now()


class PARAMETER:
    """
    项目参数
    """
    epochs = 10,
    learn_rate = 1e-3
    batch_size = 32
    PADDING_ID = -100
